How to write a function question program in hacker rank